// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_RENDERER_P2P_PORT_ALLOCATOR_H_
#define CONTENT_RENDERER_P2P_PORT_ALLOCATOR_H_

#include <memory>

#include "base/macros.h"
#include "base/memory/ref_counted.h"
#include "base/single_thread_task_runner.h"
#include "third_party/webrtc/p2p/client/basicportallocator.h"
#include "url/gurl.h"

namespace content {

class P2PSocketDispatcher;

class P2PPortAllocator : public cricket::BasicPortAllocator {
public:
    struct Config {
        // Enable non-proxied UDP-based transport when set to true. When set to
        // false, it effectively disables all UDP traffic until UDP-supporting proxy
        // RETURN is available in the future.
        bool enable_nonproxied_udp = true;

        // Request binding to individual NICs. Whether multiple routes is allowed is
        // subject to the permission check on mic/camera. When specified as false or
        // the permission request is denied, it still uses the default local address
        // to generate a single local candidate. TODO(guoweis): Rename this to
        // |request_multiple_routes|.
        bool enable_multiple_routes = true;

        // Enable exposing the default local address when set to true. This is
        // only in effect when the |enable_multiple_routes| is false or the
        // permission check of mic/camera is denied.
        bool enable_default_local_candidate = true;
    };

    P2PPortAllocator(const scoped_refptr<P2PSocketDispatcher>& socket_dispatcher,
        std::unique_ptr<rtc::NetworkManager> network_manager,
        rtc::PacketSocketFactory* socket_factory,
        const Config& config,
        const GURL& origin);
    ~P2PPortAllocator() override;

    // Will also initialize the network manager passed into the constructor.
    void Initialize() override;

private:
    std::unique_ptr<rtc::NetworkManager> network_manager_;
    scoped_refptr<P2PSocketDispatcher> socket_dispatcher_;
    Config config_;
    GURL origin_;

    DISALLOW_COPY_AND_ASSIGN(P2PPortAllocator);
};

} // namespace content

#endif // CONTENT_RENDERER_P2P_PORT_ALLOCATOR_H_
